from ultralytics import YOLO

# Load a pretrained YOLO11n model
model = YOLO("/workspace/ultralytics/models/yolov8s.pt")

# Run inference on 'bus.jpg' with arguments
model.predict("/workspace/ultralytics/ultralytics_mlu/ultralytics/assets/bus.jpg", save=True, imgsz=320, conf=0.5)
